登录返回隔离点注册信息

FFIB 2 年 前
コミット
5f750e4cae
共有2 個のファイルを変更した16 個の追加0 個の削除を含む
  1. 8 0
      account/models.py
  2. 8 0
      equipment/models.py

+ 8 - 0
account/models.py

@@ -5,6 +5,8 @@ from django.utils.translation import ugettext_lazy as _
5 5
 from django_models_ext import BaseModelMixin, SexModelMixin
6 6
 from shortuuidfield import ShortUUIDField
7 7
 
8
+from equipment.models import IsolationPointUserInfo
9
+
8 10
 
9 11
 class UserInfo(BaseModelMixin):
10 12
     UNVERIFIED = 0
@@ -49,6 +51,11 @@ class UserInfo(BaseModelMixin):
49 51
 
50 52
     @property
51 53
     def data(self):
54
+        try:
55
+            user = IsolationPointUserInfo.objects.get(user_id=self.user_id, status=True).userdata
56
+        except:
57
+            user = {}
58
+
52 59
         return {
53 60
             'user_id': self.user_id,
54 61
             'unionid': self.unionid,
@@ -63,6 +70,7 @@ class UserInfo(BaseModelMixin):
63 70
             'province': self.province,
64 71
             'city': self.city,
65 72
             'user_status': self.user_status,
73
+            'point_info': user
66 74
         }
67 75
 
68 76
 

+ 8 - 0
equipment/models.py

@@ -186,6 +186,14 @@ class IsolationPointUserInfo(BaseModelMixin):
186 186
             'remark': self.remark or '',
187 187
         }
188 188
 
189
+    @property
190
+    def userdata(self):
191
+        return {
192
+           'point_id': self.point_id,
193
+           'user_id': self.user_id,
194
+           **{ field.get('key', ''): field.get('value', '') for field in self.fields }
195
+        }
196
+
189 197
 
190 198
 class ThermometerEquipmentInfo(BaseModelMixin):
191 199
     ONLINE = 1